package com.song.controller.admin;

import com.alibaba.fastjson.JSON;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.song.pojo.Menu;
import com.song.pojo.Role;
import com.song.service.EmployeeService;
import com.song.service.MenuService;
import com.song.service.RoleService;
import com.song.utils.DataGridViewResult;
import com.song.utils.SystemConstant;
import com.song.utils.TreeNode;
import com.song.vo.RoleVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author 宋超
 * HP - the login name of the current user.
 * 2020/12/10 - the current system date.
 * 21:55 - the current system time.
 * 角色管理
 */
@RequestMapping("/admin/role")
@RestController
public class RoleController {


    @Autowired
    private RoleService roleService;

    @Autowired
    private EmployeeService employeeService;

    @Autowired
    private MenuService menuService;

    /**
     * 角色列表查询
     * @return
     */
    @RequestMapping("/list")
    public DataGridViewResult list(RoleVo roleVo){
        //设置分页信息
        PageHelper.startPage(roleVo.getPage(),roleVo.getLimit());
        //调用分页查询的方法
        List<Role> roleList = roleService.findRoleList(roleVo);
        //创建分页对象
        PageInfo<Role> pageInfo = new PageInfo<Role>(roleList);
        //返回数据
        return new DataGridViewResult(pageInfo.getTotal(),pageInfo.getList());
    }



    //添加部门信息
    @RequestMapping("/addRole")
    public String addRole(Role role){
        //创建map存放信息
        Map<String,Object> map = new HashMap<String, Object>();
        int addRole = roleService.addRole(role);
        if(addRole>0){
            map.put(SystemConstant.SUCCESS,true);
            map.put(SystemConstant.MESSAGE,"添加成功");
        }else{
            map.put(SystemConstant.SUCCESS,false);
            map.put(SystemConstant.MESSAGE,"添加失败");
        }
        //map中存放的信息已json形式返回
        return JSON.toJSONString(map);
    }

    //修改角色信息
    @RequestMapping("/updateRole")
    public String updateRole(Role role){
        Map<String,Object> map = new HashMap<String, Object>();
        int updateRole = roleService.updateRole(role);
        if(updateRole>0){
            map.put(SystemConstant.SUCCESS,true);
            map.put(SystemConstant.MESSAGE,"修改成功");
        }else{
            map.put(SystemConstant.SUCCESS,false);
            map.put(SystemConstant.MESSAGE,"修改失败");
        }
        //map中存放的信息已json形式返回
        return JSON.toJSONString(map);
    }

    //删除角色信息
    @RequestMapping("/deleteById")
    public String deleteById(Integer id){
        Map<String,Object> map = new HashMap<String, Object>();
        int deleteById = roleService.deleteById(id);
        if (deleteById>0){
            map.put(SystemConstant.SUCCESS,true);//删除成功
            map.put(SystemConstant.MESSAGE,"删除成功");
        }else{
            map.put(SystemConstant.SUCCESS,false);//删除失败
            map.put(SystemConstant.MESSAGE,"删除失败");
        }
        return JSON.toJSONString(map);
    }


    //查看该角色编号下是否有员工
    @RequestMapping("/checkRoleHasEmployee")
    public String checkDeptHasEmployee(Integer id){
        Map<String,Object> map = new HashMap<String, Object>();
        int countById = employeeService.getEmployeeCountByRoleId(id);
        if (countById>0){
            //该角色是否有员工在用
            map.put(SystemConstant.EXISI,true);//存在
            map.put(SystemConstant.MESSAGE,"该角色有员工使用,不能删除");
        }else{
            map.put(SystemConstant.EXISI,false);//存在
        }
        return JSON.toJSONString(map);
    }


    /**
     * 初始化菜单
     * @return
     */
    @RequestMapping("/initMenuTree")
    public DataGridViewResult initMenuTree(Integer roleId){
        //调用查询菜单列表的方法
        List<Menu> menuList = menuService.findMenuList();

        //根据角色ID查询该角色已经拥有的菜单ID的方法
        List<Integer> currentRoleMenuIds = menuService.findMenuIdListByRoleId(roleId);
        //定义集合，保存菜单信息
        List<Menu> currentMenus = new ArrayList<Menu>();
        //判断集合是否存在数据
        if(currentRoleMenuIds!=null && currentRoleMenuIds.size()>0){
        //根据菜单ID查询该菜单的信息
            currentMenus = menuService.findMenuByMenuId(currentRoleMenuIds);
        }
        //创建集合保存树节点信息
        List<TreeNode> treeNodes = new ArrayList<TreeNode>();
        //循环遍历集合
        for (Menu menu : menuList) {
            //定义变量，标识是否选中
            String checkArr = "0";//0表示复选框不选中，1表示选中复选框
            //内层循环遍历当前角色拥有的权限菜单
            //循环比较的原因：比较两个集合中的数据是否有相同的，有相同的表示当前角色拥有这个权限
            for (Menu currentMenu : currentMenus) {
                //如果ID相等，表示当前角色有这个菜单，有这个菜单则需要将复选框选中
                if(menu.getId() == currentMenu.getId()){
                    checkArr ="1";//选中
                    break;
                }
            }

            //定义变量，表示菜单是否展开
            Boolean spread = (menu.getSpread()==null || menu.getSpread()==1) ? true : false;
            treeNodes.add(new TreeNode(menu.getId(),menu.getPid(),menu.getTitle(),spread,checkArr));
        }
        //将数据返回到页面
        return new DataGridViewResult(treeNodes);
    }

    /**
     * 分配菜单
     * @param ids
     * @param roleId
     * @return
     */
    @RequestMapping("/saveRoleMenu")
    public String saveRoleMenu(String ids,Integer roleId){
        Map<String,Object> map = new HashMap<String,Object>();
        //调用保存角色菜单关系的方法
        if(roleService.saveRoleMenu(ids,roleId)>0){
            map.put(SystemConstant.MESSAGE,"菜单分配成功");
        }else{
            map.put(SystemConstant.MESSAGE,"菜单分配失败");
        }
        return JSON.toJSONString(map);
    }


    /**
     * 查询角色列表通过员工ID
     * @param id
     * @return
     */
    @RequestMapping("/initRoleListByEmpId")
    public DataGridViewResult initRoleListByEmpId(int id){
        List<Map<String, Object>> roleList = roleService.findRoleListByMap();
        //调用根据员工ID查询该员工拥有的角色列表的方法
        List<Integer> roleIds = roleService.findEmployeeRoleByEmployeeId(id);
        //循环比较两个集合中的角色ID值是否相等，相等则选中该角色
        for (Map<String, Object> map : roleList) {
            //定义变量，标识是否选中
            boolean flag = false;//不选中
            //获取每一个角色ID
            Integer rid = (Integer) map.get("id");//id是主键，以主键作为map集合中key
            //内层循环遍历该员工拥有的角色列表
            for (Integer roleId : roleIds) {
                //判断两个集合中是否存在角色ID相同
                if(rid == roleId){
                    flag = true;//选中该角色
                    break;
                }
            }
            //将状态保存在map集合中
            map.put("LAY_CHECKED",flag);//key必须是LAY_CHECKED
        }
        return new DataGridViewResult(Long.valueOf(roleList.size()),roleList);
    }
}
